package com.yrihr.wisdom.util;

import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;

import java.io.IOException;
import java.net.HttpURLConnection;
import java.net.URL;

/**
 * @ClassName : JsoupUtil
 * @Description : 封装jsoup连接数据网站
 * @Author : wind
 * @Date: 2020-11-16 17:12
 * @Version 1.0
 */
public class JsoupUtil {

    /**
     * @param url         请求地址
     * @param method      请求方法类型
     * @param charsetName 编码方式
     * @return
     * @throws IOException
     */
    public static Document getConnection(String url, String method, String charsetName) throws IOException {
        URL conUrl = new URL(url);
        HttpURLConnection connection = (HttpURLConnection) conUrl.openConnection();
        // 设置请求方法类型
        connection.setRequestMethod(method);
        // 是否允许缓存
        connection.setUseCaches(false);
        // 设置请求头信息
        connection.addRequestProperty("Connection", "close");
        // 设置连接主机超时（单位：毫秒）
        connection.setConnectTimeout(80000);
        // 设置从主机读取数据超时（单位：毫秒）
        connection.setReadTimeout(80000);
        // 开始请求
        Document doc = null;
        try {
            doc = Jsoup.parse(connection.getInputStream(), charsetName, "");
            connection.disconnect();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return doc;
    }
}
